from .control import Control


class Tree(Control):
    '''
    树形控件操作
    '''

    def collapse_all(self):
        '''
        全部折叠
        '''
        self.control.CollapseAll

    def expand_all(self):
        '''
        全部展开
        '''
        self.control.ExpandAll

    def get_immediate_items_text(self):
        '''
        获取直接子节点文本集合

        :Returns:
            返回子节点文本集合
        '''
        return self.control.GetImmediateItemsText

    def get_immediate_selected_items_text(self):
        '''
        获取选中的直接子节点文本集合

        :Returns:
            返回选中的子节点文本集合
        '''
        return self.control.GetImmediateSelectedItemsText




